package database

import (
	"fmt"
	"time"

	"gitee.com/botaoshow/yuanxin/web/common/conf"

	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
	"gorm.io/gorm/schema"
)

func InitMysql() {
	cfg := conf.AppConfig.Database
	// 使用配置文件中的数据库配置
	dsn := conf.AppConfig.Database.Username + ":" +
		cfg.Password + "@(" +
		cfg.Host + ":" +
		cfg.Port + ")/" +
		cfg.Database +
		"?charset=" + cfg.Charset +
		"&parseTime=True&loc=Local"

	db, err := gorm.Open(mysql.New(mysql.Config{
		DSN:                       dsn,   // DSN data source name
		DefaultStringSize:         256,   // string 类型字段的默认长度
		DisableDatetimePrecision:  true,  // 禁用 datetime 精度，MySQL 5.6 之前的数据库不支持
		DontSupportRenameIndex:    true,  // 重命名索引时采用删除并新建的方式，MySQL 5.7 之前的数据库和 MariaDB 不支持重命名索引
		DontSupportRenameColumn:   true,  // 用 `change` 重命名列，MySQL 8 之前的数据库和 MariaDB 不支持重命名列
		SkipInitializeWithVersion: false, // 根据当前 MySQL 版本自动配置
	}), &gorm.Config{
		NamingStrategy: schema.NamingStrategy{
			SingularTable: true, // 使用单数表名，启用该选项，此时，`User` 的表名应该是 `user`
		},
		//打印日志
		Logger: logger.Default.LogMode(logger.Info),
	})
	if err != nil {
		panic(err)
	}
	sqlDB, err := db.DB()

	// SetMaxIdleConns 设置空闲连接池中连接的最大数量。
	sqlDB.SetMaxIdleConns(*cfg.MaxIdleCount)

	// SetMaxOpenConns 设置打开数据库连接的最大数量。
	sqlDB.SetMaxOpenConns(*cfg.MaxOpenConns)

	// SetConnMaxLifetime 设置了可以重新使用连接的最大时间。
	lifetime := time.Duration(*cfg.ConnMaxLifetime) * time.Minute
	sqlDB.SetConnMaxLifetime(lifetime)
	Db = db
	fmt.Println("mysql database connected")
}
func CloseMysql() {
	sqlDB, _ := Db.DB()
	err := sqlDB.Close()
	if err != nil {
		return
	}
	Db = nil
	fmt.Println("mysql database closed")
}
